# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     config
   Description :   
   Author :       lth
   date：          2023/1/30
-------------------------------------------------
   Change Activity:
                   2023/1/30 10:28: create this script
-------------------------------------------------
"""
__author__ = 'lth'


class GetConfig:
    project_name = "text classification"
    use_cuda = True
    seed = 123
    resume = True
    pretrained_weights = "./weights/best.pth"
    lr = 0.001
    epochs = 200

    train_batch_size = 16
    test_batch_size = 16
